import numpy as np
import matplotlib.pyplot as plt
import random

# 绘制NISQA
# # 准备数据
# x_data = ["AdaIN", "FragmentVC","VQMIVC", "Proposed"]
# y_data = [2.99, 3.03, 3.12, 3.38]

# # 正确显示中文和负号
# plt.rcParams["font.sans-serif"] = ["SimHei"]
# plt.rcParams["axes.unicode_minus"] = False

# # 画图，plt.bar()可以画柱状图
# for i in range(len(x_data)):
# 	plt.bar(x_data[i], y_data[i])
# # 设置图片名称
# # plt.title("销量分析")
# # 设置x轴标签名
# plt.xlabel("Model")
# # 设置y轴标签名
# plt.ylabel("Mos")
# # 显示
# plt.show()

# plt.savefig(fname="./test.png", dpi=100)

# 绘制WER
# 准备数据
x_data = ["AdaIN", "FragmentVC","VQMIVC", "Proposed"]
y_data = [47.3, 36.4, 27.8, 15.6]

# 正确显示中文和负号
plt.rcParams["font.sans-serif"] = ["SimHei"]
plt.rcParams["axes.unicode_minus"] = False

# 画图，plt.bar()可以画柱状图
for i in range(len(x_data)):
	plt.bar(x_data[i], y_data[i])
# 设置图片名称
# plt.title("销量分析")
# 设置x轴标签名
plt.xlabel("Model")
# 设置y轴标签名
plt.ylabel("WER")
# 显示
plt.show()

plt.savefig(fname="./wer.png", dpi=100)
